package umage.projects;

import java.util.ArrayList;
import java.util.List;

import umage.action.DefaultAction;
import umage.action.ClassExportAction;
import umage.action.ClassImportAction;
import umage.figures.*;

import org.jhotdraw.draw.Figure;

@ProjectInfo(name = "Class Diagram Project", icon = "ClassDiagramProject.png")
public class ClassDiagramProject extends UMageProject {
	
	@Override
	public List<Class<? extends Figure>> getNeededFigures() {
		ArrayList<Class<? extends Figure>> figures = new ArrayList<Class<? extends Figure>>();
		figures.add(ClassFigure.class);
		figures.add(InterfaceFigure.class);
		figures.add(AggregationArrow.class);
		figures.add(AssociationArrow.class);
		figures.add(CompositionArrow.class);
		figures.add(DependencyArrow.class);
		figures.add(GeneralizationArrow.class);
		figures.add(ImplementsArrow.class);
		return figures;
	}
	
	@Override
	public List<Class<? extends DefaultAction>> getNeededActions() {
		ArrayList<Class<? extends DefaultAction>> actions = new ArrayList<Class<? extends DefaultAction>>();
		actions.add(ClassImportAction.class);
		actions.add(ClassExportAction.class);
		return actions;
	}
	
}
